import warnings
warnings.filterwarnings('ignore')
from ultralytics import YOLO

# if __name__ == '__main__':
#     model = YOLO('ricepest_16/yolov8n_cbam_qfl0.25_91.5.pt') # select your model.pt path
#     model.predict(source='D:/test/images',
#                 project='runs/detect',
#                 name='exp',
#                 save=True,
#                 # visualize=True # visualize model features maps
#                 )

if __name__ == '__main__':
    model = YOLO('pest24/yolov8s.pt') # select your model.pt path
    model.predict(source='pest24/0000169.jpg',
                project='runs/detect',
                name='exp',
                save=True,
                # visualize=True # visualize model features maps
                )


